How to get list as user input in python